import wbdata
import sqlite3
import pandas as pd
from datetime import datetime

# 设置开始和结束日期
start_date = datetime(2018, 1, 1)
end_date = datetime(2025, 1, 1)

# 选择需要的国家
# countries = ['US', 'CN', 'IN', 'BR', 'RU', 'DE', 'JP']
countries = ['CN','JP']

# 选择需要的经济指标
indicators = {
    'BG.GSR.NFSV.GD.ZS':'服务贸易GDP%',
    'NY.GDP.MKTP.CD': 'GDP',
    'FP.CPI.TOTL.ZG': '通货膨胀率',
    'SL.UEM.TOTL.ZS': '失业率',
    'NY.GDP.PCAP.CD': '人均GDP',
    'FI.RES.TOTL.CD': '外汇储备',
    'NE.EXP.GNFS.CD': '贸易差额',
    'NE.EXP.GNFS.CD': '出口总额 (USD)',
    'NE.IMP.GNFS.CD': '进口总额 (USD)',
    'NE.RSB.GNFS.CD': '贸易差额 (USD)',
    'NE.EXP.GNFS.ZS': '出口占GDP比重 (%)',
    'NE.IMP.GNFS.ZS': '进口占GDP比重 (%)'
}

# 获取数据
data = wbdata.get_dataframe(indicators, country=countries, date=(start_date, end_date))

# 打印数据
print(data)
with sqlite3.connect("example.db") as conn: data.to_sql("wbdata_test1",conn,if_exists="replace",index=False)


